/**
 * Copyright (c) 2016-2019 人人开源 All rights reserved.
 *
 * https://www.renren.io
 *
 * 版权所有，侵权必究！
 */

package com.manli.generator.controller;


import cn.smallbun.screw.core.Configuration;
import cn.smallbun.screw.core.engine.EngineConfig;
import cn.smallbun.screw.core.engine.EngineFileType;
import cn.smallbun.screw.core.engine.EngineTemplateType;
import cn.smallbun.screw.core.execute.DocumentationExecute;
import cn.smallbun.screw.core.process.ProcessConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.sql.DataSource;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 代码生成器
 *
 *
 */
@Controller
@RequestMapping("/dbswagger/")
public class DbSwaggerController {
	@Autowired
	ApplicationContext applicationContext;
	/**
	 * 列表
	 */
	@RequestMapping("/start")
	public void start(){
		DataSource dataSourceMysql = applicationContext.getBean(DataSource.class);
		// 生成文件配置
		EngineConfig engineConfig = EngineConfig.builder()
				// 生成文件路径，自己mac本地的地址，这里需要自己更换下路径
				.fileOutputDir("D:/")
				// 打开目录
				.openOutputDir(true)
				// 文件类型
				.fileType(EngineFileType.HTML)
				// 生成模板实现
				.produceType(EngineTemplateType.freemarker).build();
		// 生成文档配置（包含以下自定义版本号、描述等配置连接）
		Configuration config = Configuration.builder()
				.version("1.0.3")
				.description("生成文档信息描述")
				.dataSource(dataSourceMysql)
				.engineConfig(engineConfig)
				.produceConfig(getProcessConfig())
				.build();
		// 执行生成
		new DocumentationExecute(config).execute();
	}

	/**
	 * 配置想要生成的表+ 配置想要忽略的表
	 *
	 * @return 生成表配置
	 */
	public static ProcessConfig getProcessConfig() {
		// 忽略表名
		List<String> ignoreTableName = Arrays.asList();
		// 忽略表前缀，如忽略a开头的数据库表
		List<String> ignorePrefix = Arrays.asList();
		// 忽略表后缀
		List<String> ignoreSuffix = Arrays.asList();
		return ProcessConfig.builder()
				//根据名称指定表生成
				//.designatedTableName(Arrays.asList("fire_user"))
				//根据表前缀生成
				//.designatedTablePrefix(new ArrayList<>())
				//根据表后缀生成
				//.designatedTableSuffix(new ArrayList<>())
				//忽略表名
				.ignoreTableName(ignoreTableName)
				//忽略表前缀
				.ignoreTablePrefix(ignorePrefix)
				//忽略表后缀
				.ignoreTableSuffix(ignoreSuffix)
				.build();
	}
	

}
